home *** CD-ROM | disk | FTP | other *** search
/ Die Speccy' 97 / Die Speccy' 97.iso / amiga_system / the_aminet / dev / lang / python_src.lha / amigapython / objects / object.c < prev    next >
C/C++ Source or Header  |  1995-10-22  |  12KB  |  575 lines

  1. /***********************************************************
  2. Copyright 1991-1995 by Stichting Mathematisch Centrum, Amsterdam,
  3. The Netherlands.
  4.  
  5.                         All Rights Reserved
  6.  
  7. Permission to use, copy, modify, and distribute this software and its 
  8. documentation for any purpose and without fee is hereby granted, 
  9. provided that the above copyright notice appear in all copies and that
  10. both that copyright notice and this permission notice appear in 
  11. supporting documentation, and that the names of Stichting Mathematisch
  12. Centrum or CWI not be used in advertising or publicity pertaining to
  13. distribution of the software without specific, written prior permission.
  14.  
  15. STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO
  16. THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
  17. FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE
  18. FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  19. WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  20. ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
  21. OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  22.  
  23. ******************************************************************/
  24.  
  25. /* Generic object operations; and implementation of None (NoObject) */
  26.  
  27. #include "allobjects.h"
  28.  
  29. #if defined( Py_TRACE_REFS ) || defined( Py_REF_DEBUG )
  30. long ref_total;
  31. #endif
  32.  
  33. /* Object allocation routines used by NEWOBJ and NEWVAROBJ macros.
  34.    These are used by the individual routines for object creation.
  35.    Do not call them otherwise, they do not initialize the object! */
  36.  
  37. #ifdef COUNT_ALLOCS
  38. static typeobject *type_list;
  39. extern int tuple_zero_allocs, fast_tuple_allocs;
  40. extern int quick_int_allocs, quick_neg_int_allocs;
  41. extern int null_strings, one_strings;
  42. void
  43. dump_counts()
  44. {
  45.     typeobject *tp;
  46.  
  47.     for (tp = type_list; tp; tp = tp->tp_next)
  48.         fprintf(stderr, "%s alloc'd: %d, freed: %d, max in use: %d\n",
  49.             tp->tp_name, tp->tp_alloc, tp->tp_free,
  50.             tp->tp_maxalloc);
  51.     fprintf(stderr, "fast tuple allocs: %d, empty: %d\n",
  52.         fast_tuple_allocs, tuple_zero_allocs);
  53.     fprintf(stderr, "fast int allocs: pos: %d, neg: %d\n",
  54.         quick_int_allocs, quick_neg_int_allocs);
  55.     fprintf(stderr, "null strings: %d, 1-strings: %d\n",
  56.         null_strings, one_strings);
  57. }
  58.  
  59. PyObject *
  60. get_counts()
  61. {
  62.     PyTypeObject *tp;
  63.     PyObject *result;
  64.     PyObject *v;
  65.  
  66.     result = PyList_New(0);
  67.     if (result == NULL)
  68.         return NULL;
  69.     for (tp = type_list; tp; tp = tp->tp_next) {
  70.         v = Py_BuildValue("(siii)", tp->tp_name, tp->tp_alloc,
  71.                   tp->tp_free, tp->tp_maxalloc);
  72.         if (v == NULL) {
  73.             Py_DECREF(result);
  74.             return NULL;
  75.         }
  76.         if (PyList_Append(result, v) < 0) {
  77.             Py_DECREF(v);
  78.             Py_DECREF(result);
  79.             return NULL;
  80.         }
  81.         Py_DECREF(v);
  82.     }
  83.     return result;
  84. }
  85.  
  86. void
  87. inc_count(tp)
  88.     typeobject *tp;
  89. {
  90.     if (tp->tp_alloc == 0) {
  91.         /* first time; insert in linked list */
  92.         if (tp->tp_next != NULL) /* sanity check */
  93.             fatal("XXX inc_count sanity check");
  94.         tp->tp_next = type_list;
  95.         type_list = tp;
  96.     }
  97.     tp->tp_alloc++;
  98.     if (tp->tp_alloc - tp->tp_free > tp->tp_maxalloc)
  99.         tp->tp_maxalloc = tp->tp_alloc - tp->tp_free;
  100. }
  101. #endif
  102.  
  103. object *
  104. newobject(tp)
  105.     typeobject *tp;
  106. {
  107.     object *op = (object *) malloc(tp->tp_basicsize);
  108.     if (op == NULL)
  109.         return err_nomem();
  110.     op->ob_type = tp;
  111.     NEWREF(op);
  112.     return op;
  113. }
  114.  
  115. varobject *
  116. newvarobject(tp, size)
  117.     typeobject *tp;
  118.     int size;
  119. {
  120.     varobject *op = (varobject *)
  121.         malloc(tp->tp_basicsize + size * tp->tp_itemsize);
  122.     if (op == NULL)
  123.         return (varobject *)err_nomem();
  124.     op->ob_type = tp;
  125.     op->ob_size = size;
  126.     NEWREF(op);
  127.     return op;
  128. }
  129.  
  130. int
  131. printobject(op, fp, flags)
  132.     object *op;
  133.     FILE *fp;
  134.     int flags;
  135. {
  136.     int ret = 0;
  137.     if (sigcheck())
  138.         return -1;
  139.     if (op == NULL) {
  140.         fprintf(fp, "<nil>");
  141.     }
  142.     else {
  143.         if (op->ob_refcnt <= 0)
  144.             fprintf(fp, "<refcnt %u at %lx>",
  145.                 op->ob_refcnt, (long)op);
  146.         else if (op->ob_type->tp_print == NULL) {
  147.             if (op->ob_type->tp_repr == NULL) {
  148.                 fprintf(fp, "<%s object at %lx>",
  149.                     op->ob_type->tp_name, (long)op);
  150.             }
  151.             else {
  152.                 object *s;
  153.                 if (flags & PRINT_RAW)
  154.                     s = strobject(op);
  155.                 else
  156.                     s = reprobject(op);
  157.                 if (s == NULL)
  158.                     ret = -1;
  159.                 else if (!is_stringobject(s)) {
  160.                     err_setstr(TypeError,
  161.                            "repr not string");
  162.                     ret = -1;
  163.                 }
  164.                 else {
  165.                     fprintf(fp, "%s", getstringvalue(s));
  166.                 }
  167.                 XDECREF(s);
  168.             }
  169.         }
  170.         else
  171.             ret = (*op->ob_type->tp_print)(op, fp, flags);
  172.     }
  173.     if (ret == 0) {
  174.         if (ferror(fp)) {
  175.             err_errno(IOError);
  176.             clearerr(fp);
  177.             ret = -1;
  178.         }
  179.     }
  180.     return ret;
  181. }
  182.  
  183. object *
  184. reprobject(v)
  185.     object *v;
  186. {
  187.     if (sigcheck())
  188.         return NULL;
  189.     if (v == NULL)
  190.         return newstringobject("<NULL>");
  191.     else if (v->ob_type->tp_repr == NULL) {
  192.         char buf[120];
  193.         sprintf(buf, "<%.80s object at %lx>",
  194.             v->ob_type->tp_name, (long)v);
  195.         return newstringobject(buf);
  196.     }
  197.     else
  198.         return (*v->ob_type->tp_repr)(v);
  199. }
  200.  
  201. object *
  202. strobject(v)
  203.     object *v;
  204. {
  205.     if (v == NULL)
  206.         return newstringobject("<NULL>");
  207.     else if (is_stringobject(v)) {
  208.         INCREF(v);
  209.         return v;
  210.     }
  211.     else if (v->ob_type->tp_str != NULL)
  212.         return (*v->ob_type->tp_str)(v);
  213.     else {
  214.         object *func;
  215.         object *res;
  216.         if (!is_instanceobject(v) ||
  217.             (func = getattr(v, "__str__")) == NULL) {
  218.             err_clear();
  219.             return reprobject(v);
  220.         }
  221.         res = call_object(func, (object *)NULL);
  222.         DECREF(func);
  223.         return res;
  224.     }
  225. }
  226.  
  227. static object *
  228. do_cmp(v, w)
  229.     object *v, *w;
  230. {
  231.     /* __rcmp__ actually won't be called unless __cmp__ isn't defined,
  232.        because the check in cmpobject() reverses the objects first.
  233.        This is intentional -- it makes no sense to define cmp(x,y) different
  234.        than -cmp(y,x). */
  235.     if (is_instanceobject(v) || is_instanceobject(w))
  236.         return instancebinop(v, w, "__cmp__", "__rcmp__", do_cmp);
  237.     return newintobject((long)cmpobject(v, w));
  238. }
  239.  
  240. int
  241. cmpobject(v, w)
  242.     object *v, *w;
  243. {
  244.     typeobject *tp;
  245.     if (v == w)
  246.         return 0;
  247.     if (v == NULL)
  248.         return -1;
  249.     if (w == NULL)
  250.         return 1;
  251.     if (is_instanceobject(v) || is_instanceobject(w)) {
  252.         object *res;
  253.         int c;
  254.         if (!is_instanceobject(v))
  255.             return -cmpobject(w, v);
  256.         res = do_cmp(v, w);
  257.         if (res == NULL) {
  258.             err_clear();
  259.             return (v < w) ? -1 : 1;
  260.         }
  261.         if (!is_intobject(res)) {
  262.             DECREF(res);
  263.             return (v < w) ? -1 : 1;
  264.         }
  265.         c = getintvalue(res);
  266.         DECREF(res);
  267.         return (c < 0) ? -1 : (c > 0) ? 1 : 0;    
  268.     }
  269.     if ((tp = v->ob_type) != w->ob_type) {
  270.         if (tp->tp_as_number != NULL &&
  271.                 w->ob_type->tp_as_number != NULL) {
  272.             if (coerce(&v, &w) != 0) {
  273.                 err_clear();
  274.                 /* XXX Should report the error,
  275.                    XXX but the interface isn't there... */
  276.             }
  277.             else {
  278.                 int cmp = (*v->ob_type->tp_compare)(v, w);
  279.                 DECREF(v);
  280.                 DECREF(w);
  281.                 return cmp;
  282.             }
  283.         }
  284.         return strcmp(tp->tp_name, w->ob_type->tp_name);
  285.     }
  286.     if (tp->tp_compare == NULL)
  287.         return (v < w) ? -1 : 1;
  288.     return (*tp->tp_compare)(v, w);
  289. }
  290.  
  291. long
  292. hashobject(v)
  293.     object *v;
  294. {
  295.     typeobject *tp = v->ob_type;
  296.     if (tp->tp_hash != NULL)
  297.         return (*tp->tp_hash)(v);
  298.     if (tp->tp_compare == NULL)
  299.         return (long) v; /* Use address as hash value */
  300.     /* If there's a cmp but no hash defined, the object can't be hashed */
  301.     err_setstr(TypeError, "unhashable type");
  302.     return -1;
  303. }
  304.  
  305. object *
  306. getattr(v, name)
  307.     object *v;
  308.     char *name;
  309. {
  310.     if (v->ob_type->tp_getattr == NULL) {
  311.         err_setstr(AttributeError, "attribute-less object");
  312.         return NULL;
  313.     }
  314.     else {
  315.         return (*v->ob_type->tp_getattr)(v, name);
  316.     }
  317. }
  318.  
  319. int
  320. hasattr(v, name)
  321.     object *v;
  322.     char *name;
  323. {
  324.     object *res = getattr(v, name);
  325.     if (res != NULL) {
  326.         DECREF(res);
  327.         return 1;
  328.     }
  329.     err_clear();
  330.     return 0;
  331. }
  332.  
  333. int
  334. setattr(v, name, w)
  335.     object *v;
  336.     char *name;
  337.     object *w;
  338. {
  339.     if (v->ob_type->tp_setattr == NULL) {
  340.         if (v->ob_type->tp_getattr == NULL)
  341.             err_setstr(TypeError,
  342.                    "attribute-less object (assign or del)");
  343.         else
  344.             err_setstr(TypeError,
  345.                    "object has read-only attributes");
  346.         return -1;
  347.     }
  348.     else {
  349.         return (*v->ob_type->tp_setattr)(v, name, w);
  350.     }
  351. }
  352.  
  353. /* Test a value used as condition, e.g., in a for or if statement.
  354.    Return -1 if an error occurred */
  355.  
  356. int
  357. testbool(v)
  358.     object *v;
  359. {
  360.     int res;
  361.     if (v == None)
  362.         res = 0;
  363.     else if (v->ob_type->tp_as_number != NULL)
  364.         res = (*v->ob_type->tp_as_number->nb_nonzero)(v);
  365.     else if (v->ob_type->tp_as_mapping != NULL)
  366.         res = (*v->ob_type->tp_as_mapping->mp_length)(v);
  367.     else if (v->ob_type->tp_as_sequence != NULL)
  368.         res = (*v->ob_type->tp_as_sequence->sq_length)(v);
  369.     else
  370.         res = 1;
  371.     if (res > 0)
  372.         res = 1;
  373.     return res;
  374. }
  375.  
  376. /* Coerce two numeric types to the "larger" one.
  377.    Increment the reference count on each argument.
  378.    Return -1 and raise an exception if no coercion is possible
  379.    (and then no reference count is incremented).
  380. */
  381.  
  382. int
  383. coerce(pv, pw)
  384.     object **pv, **pw;
  385. {
  386.     register object *v = *pv;
  387.     register object *w = *pw;
  388.     int res;
  389.  
  390.     if (v->ob_type == w->ob_type && !is_instanceobject(v)) {
  391.         INCREF(v);
  392.         INCREF(w);
  393.         return 0;
  394.     }
  395.     if (v->ob_type->tp_as_number && v->ob_type->tp_as_number->nb_coerce) {
  396.         res = (*v->ob_type->tp_as_number->nb_coerce)(pv, pw);
  397.         if (res <= 0)
  398.             return res;
  399.     }
  400.     if (w->ob_type->tp_as_number && w->ob_type->tp_as_number->nb_coerce) {
  401.         res = (*w->ob_type->tp_as_number->nb_coerce)(pw, pv);
  402.         if (res <= 0)
  403.             return res;
  404.     }
  405.     err_setstr(TypeError, "number coercion failed");
  406.     return -1;
  407. }
  408.  
  409.  
  410. /* Test whether an object can be called */
  411.  
  412. int
  413. callable(x)
  414.     object *x;
  415. {
  416.     if (x == NULL)
  417.         return 0;
  418.     if (x->ob_type->tp_call != NULL ||
  419.         is_funcobject(x) ||
  420.         is_instancemethodobject(x) ||
  421.         is_methodobject(x) ||
  422.         is_classobject(x))
  423.         return 1;
  424.     if (is_instanceobject(x)) {
  425.         object *call = getattr(x, "__call__");
  426.         if (call == NULL) {
  427.             err_clear();
  428.             return 0;
  429.         }
  430.         /* Could test recursively but don't, for fear of endless
  431.            recursion if some joker sets self.__call__ = self */
  432.         DECREF(call);
  433.         return 1;
  434.     }
  435.     return 0;
  436. }
  437.  
  438.  
  439. /*
  440. NoObject is usable as a non-NULL undefined value, used by the macro None.
  441. There is (and should be!) no way to create other objects of this type,
  442. so there is exactly one (which is indestructible, by the way).
  443. */
  444.  
  445. /* ARGSUSED */
  446. static object *
  447. none_repr(op)
  448.     object *op;
  449. {
  450.     return newstringobject("None");
  451. }
  452.  
  453. static typeobject Notype = {
  454.     OB_HEAD_INIT(&Typetype)
  455.     0,
  456.     "None",
  457.     0,
  458.     0,
  459.     0,        /*tp_dealloc*/ /*never called*/
  460.     0,        /*tp_print*/
  461.     0,        /*tp_getattr*/
  462.     0,        /*tp_setattr*/
  463.     0,        /*tp_compare*/
  464.     (reprfunc)none_repr, /*tp_repr*/
  465.     0,        /*tp_as_number*/
  466.     0,        /*tp_as_sequence*/
  467.     0,        /*tp_as_mapping*/
  468.     0,        /*tp_hash */
  469. };
  470.  
  471. object NoObject = {
  472.     OB_HEAD_INIT(&Notype)
  473. };
  474.  
  475.  
  476. #ifdef TRACE_REFS
  477.  
  478. static object refchain = {&refchain, &refchain};
  479.  
  480. NEWREF(op)
  481.     object *op;
  482. {
  483.     ref_total++;
  484.     op->ob_refcnt = 1;
  485.     op->_ob_next = refchain._ob_next;
  486.     op->_ob_prev = &refchain;
  487.     refchain._ob_next->_ob_prev = op;
  488.     refchain._ob_next = op;
  489. #ifdef COUNT_ALLOCS
  490.     inc_count(op->ob_type);
  491. #endif
  492. }
  493.  
  494. UNREF(op)
  495.     register object *op;
  496. {
  497.     register object *p;
  498.     if (op->ob_refcnt < 0)
  499.         fatal("UNREF negative refcnt");
  500.     if (op == &refchain ||
  501.         op->_ob_prev->_ob_next != op || op->_ob_next->_ob_prev != op)
  502.         fatal("UNREF invalid object");
  503. #ifdef SLOW_UNREF_CHECK
  504.     for (p = refchain._ob_next; p != &refchain; p = p->_ob_next) {
  505.         if (p == op)
  506.             break;
  507.     }
  508.     if (p == &refchain) /* Not found */
  509.         fatal("UNREF unknown object");
  510. #endif
  511.     op->_ob_next->_ob_prev = op->_ob_prev;
  512.     op->_ob_prev->_ob_next = op->_ob_next;
  513.     op->_ob_next = op->_ob_prev = NULL;
  514. #ifdef COUNT_ALLOCS
  515.     op->ob_type->tp_free++;
  516. #endif
  517. }
  518.  
  519. DELREF(op)
  520.     object *op;
  521. {
  522.     destructor dealloc = op->ob_type->tp_dealloc;
  523.     UNREF(op);
  524.     op->ob_type = NULL;
  525.     (*dealloc)(op);
  526. }
  527.  
  528. printrefs(fp)
  529.     FILE *fp;
  530. {
  531.     object *op;
  532.     fprintf(fp, "Remaining objects (except strings referenced once):\n");
  533.     for (op = refchain._ob_next; op != &refchain; op = op->_ob_next) {
  534.         if (op->ob_refcnt == 1 && is_stringobject(op))
  535.             continue; /* Will be printed elsewhere */
  536.         fprintf(fp, "[%d] ", op->ob_refcnt);
  537.         if (printobject(op, fp, 0) != 0)
  538.             err_clear();
  539.         putc('\n', fp);
  540.     }
  541. }
  542.  
  543. PyObject *
  544. getobjects(self, args)
  545.     PyObject *self;
  546.     PyObject *args;
  547. {
  548.     int i, n;
  549.     PyObject *t = NULL;
  550.     PyObject *res, *op;
  551.  
  552.     if (!PyArg_ParseTuple(args, "i|O", &n, &t))
  553.         return NULL;
  554.     op = refchain._ob_next;
  555.     res = PyList_New(0);
  556.     if (res == NULL)
  557.         return NULL;
  558.     for (i = 0; (n == 0 || i < n) && op != &refchain; i++) {
  559.         while (op == self || op == args || op == res || op == t ||
  560.                t != NULL && op->ob_type != (PyTypeObject *) t) {
  561.             op = op->_ob_next;
  562.             if (op == &refchain)
  563.                 return res;
  564.         }
  565.         if (PyList_Append(res, op) < 0) {
  566.             Py_DECREF(res);
  567.             return NULL;
  568.         }
  569.         op = op->_ob_next;
  570.     }
  571.     return res;
  572. }
  573.  
  574. #endif
  575.